# %%
from zhipuai import ZhipuAI
from docx import Document
client = ZhipuAI(api_key="") # 请填写您自己的APIKey
# %%

doc = Document('美国个人所得税法.docx')
# %%
doc.paragraphs[201].text

# %%
content = []
for idx, para in enumerate(doc.paragraphs):
    if idx > 10:
        break

    prompt = f"""
        这是出自美国个人所得税税法中的一段话，请帮我翻译为中文，如果有编号请保持编号格式，不要重复问题，直接回答: \n{para.text}\n
    """
    response= client.chat.completions.create(
        model="GLM-4-Flash-250414",  # 填写需要调用的模型编码
        messages=[
            {"role": "system", "content": "美国税制专家"}, # 大语言模型的角色
            {"role": "user", "content": prompt}    # 对该角色说的话
        ],
    )
    content.append(response.choices[0].message.content)
# %%

content = []
for para in doc.paragraphs:

    prompt = f"""
        这是出自美国个人所得税税法中的一段话，请帮我翻译为中文，如果有编号请保持编号格式，不要重复问题，直接回答: \n{para.text}\n
    """
    response= client.chat.completions.create(
        model="GLM-4-Flash-250414",  # 填写需要调用的模型编码
        messages=[
            {"role": "system", "content": "美国税制专家"}, # 大语言模型的角色
            {"role": "user", "content": prompt}    # 对该角色说的话
        ],
    )
    content.append(response.choices[0].message.content)

# %%

contents = "\n".join(content)

# %%

contents_doc = Document()
contents_doc.add_paragraph(contents)
contents_doc.save("美国个人所得税法（中文）.docx")
# %%
table_list = []
for table in doc.tables:
    for row in table.rows:
        table_list.append(" | ".join(cell.text for cell in row.cells))

# %%
#
table_contents = []
for table in table_list:

    prompt = f"""
        这是出自美国个人所得税税法中的一段话，请帮我翻译为中文，如果有编号请保持编号格式，不要重复问题，直接回答: \n{table}\n
    """
    response= client.chat.completions.create(
        model="GLM-4-Flash-250414",  # 填写需要调用的模型编码
        messages=[
            {"role": "system", "content": "美国税制专家"}, # 大语言模型的角色
            {"role": "user", "content": prompt}    # 对该角色说的话
        ],
    )
    table_contents.append(response.choices[0].message.content)

# %%
table_contents = "\n".join(table_contents)
table_contents_doc = Document()
table_contents_doc.add_paragraph(table_contents)
table_contents_doc.save("美国个人所得税法（表格）.docx")
# %%
